Machine learning screening of COVID-19 patients based on X-ray images for unbalanced classes

Background: COVID-19 is a pandemic that had already infected more than forty-six million people and caused more than a million deaths by 1st of November 2020. The virus pandemic appears to have had a catastrophic effect on the global population's safety. Therefore, efficient detection of infected patients is a key phase in the battle against COVID-19. One of the main screening methods is radiological testing. The goal of this study is using chest X-ray images to detect COVID-19 pneumonia patients while optimizing detection efficiency. Methods: As shown in Figure 1, we combined three methods to detect COVID-19 namely: convolutional neural network, transfer learning, and the focal loss1 function which are used for unbalanced classes, to build three binary classifiers which are COVID-19 versus normal, COVID-19 versus pneumonia, and COVID-19 versus normal pneumonia (normal and pneumonia). The database used2 includes a mixture of 400 COVID-19, 1,340 viral pneumonia, 2,560 bacterial pneumonia, and 1,340 normal chest X-ray images for training, validation, and testing of four pre-trained deep convolutional neural networks. Then, the pre-trained model that gives the best results was chosen to improve its performances by two enhancement techniques which are image augmentation, allowing us to reach approximately 2,500 images per class, and the adjustment of focal loss hyperparameters. Results: A comparative study was conducted of our proposed classifiers with well-known classifiers and obtained much better results in terms of accuracy, specificity, sensitivity and precision, as illustrated in Table 1. Conclusion: The high performance of this computer-aided diagnostic technique may greatly increase the screening speed and reliability of COVID-19 diagnostic cases. Particularly, at the crowded emergency services, it will be particularly helpful in this pandemic when the risk of infection and the necessity for prevention initiatives run contrary to the available resources.

Machine Learning Screening of COVID-19 Patients Based on X-ray Images for Imbalanced Classes

COVID-19 is a virus that has infected more than one hundred and fifty million people and caused more than three million deaths by 13th of Mai 2021 and is having a catastrophic effect on the world population's safety. Therefore, early detection of infected people is essential to fight this pandemic and one of the main screening methods is radiological testing. The goal of this study is the usage of chest x-ray images (CXRs) to effectively identify patients with COVID-19 pneumonia. To achieve an efficient model, we combined three methods named: Convolution Neural Network (CNN), transfer learning, and the focal loss function which is used for imbalanced classes to build 3 binary classifiers, namely COVID-19 vs Normal, COVID-19 vs pneumonia and COVID-19 vs Normal Pneumonia (Normal and Pneumonia). A comparative study has been made between our proposed classifiers with well-known classifiers and provided enhanced results in terms of accuracy, specificity, sensitivity and precision. The high performance of this computer-Aided diagnostic technique may greatly increase the screening speed and reliability of COVID-19 detection. © 2021 IEEE.